package parametrage;

import java.io.File;
import java.util.HashMap;

public class Parametrage {

	private HashMap<String, Double> param1 = new HashMap<String, Double>();
	private HashMap<String, Double> param2 = new HashMap<String, Double>();
	private int nbSpam = 0;
	private int nbHam = 0;
	
	public Parametrage() {
	
	}
	
	public void parametre(int indice, HashMap<String, Integer> beta1, HashMap<String, Integer> beta2) {
		File[] files = null;
		
		for(int i=1; i<=10; i++) {
			if(i != indice) {
			    // le repertoire à lire
				File directoryToScan = new File("/home/jerome/Bureau/TP1-EIT/lingspam/bare/part"+i); 
			    // les fichiers à lire
				files = directoryToScan.listFiles(); 
				// parcours des fichiers
				for( File fichier : files) {
			    	if(fichier != null) {
			    		if(fichier.getName().contains("spm")) {
			    			nbSpam++;
			    		}
			    		else {
			    			nbHam++;
			    		}
			    	}
				}
			}
			
			//System.out.println("Il y a "+nbHam+" hams et "+nbSpam+" spams dans le corpus");
			
			for(String s : beta1.keySet()) {
				param1.put(s, new Double((double)(beta1.get(s)+0.05)/(nbHam+0.1)));
				param2.put(s, new Double((double)(beta2.get(s)+0.05)/(nbSpam+0.1)));
				//System.out.print(s+" est dans la classe Ham dans "+beta1.get(s)+" documents ");
				//System.out.println(beta1.get(s)+"/"+nbHam+" = "+param1.get(s).doubleValue());
				//System.out.print(s+" est dans la classe Spam dans "+beta2.get(s)+" documents ");
				//System.out.println(beta2.get(s)+"/"+nbSpam+" = "+param2.get(s).doubleValue()+"\n");
			}
		}
	}

	public HashMap<String, Double> getParam1() {
		return param1;
	}

	public HashMap<String, Double> getParam2() {
		return param2;
	}
	
}
